#!/usr/bin/env python
"""
创建测试数据脚本
"""
import os
import sys
import django

# 设置Django环境
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'prompt_manager.settings')
django.setup()

from users.models import MembershipType
from subscriptions.models import PricingPlan

def create_test_data():
    """创建测试数据"""
    
    # 创建会员类型
    membership_types = [
        {
            'name': '基础版',
            'prompt_limit': 100,
            'price': 29.00,
            'description': '适合个人用户的基础功能'
        },
        {
            'name': '专业版',
            'prompt_limit': 500,
            'price': 99.00,
            'description': '适合专业用户的高级功能'
        },
        {
            'name': '企业版',
            'prompt_limit': 2000,
            'price': 299.00,
            'description': '适合企业用户的全功能版本'
        }
    ]
    
    for mt_data in membership_types:
        membership_type, created = MembershipType.objects.get_or_create(
            name=mt_data['name'],
            defaults=mt_data
        )
        if created:
            print(f"创建会员类型: {membership_type.name}")
        
        # 为每个会员类型创建价格方案
        pricing_plan, created = PricingPlan.objects.get_or_create(
            membership_type=membership_type,
            defaults={
                'price': mt_data['price'],
                'duration_days': 30,
                'description': mt_data['description'],
                'is_active': True
            }
        )
        if created:
            print(f"创建价格方案: {pricing_plan.membership_type.name} - ¥{pricing_plan.price}")

if __name__ == '__main__':
    create_test_data()
    print("测试数据创建完成！")